package com.augrain.microframework.mybatis.mapper;

import cn.augrain.easy.tool.bean.BeanMapper;
import cn.augrain.easy.tool.collection.CollectionUtils;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.ReflectionKit;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.toolkit.Db;

import java.io.Serializable;
import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 自定义 Mapper 接口, 实现 自定义扩展
 *
 * @param <T> table 泛型
 * @param <V> dto 泛型
 * @author biaoy
 * @since 2025-07-26
 */
public interface BaseMapperPlus<T, V> extends BaseMapper<T> {

    /**
     * 获取当前类的泛型类型 V 的 Class 对象
     * <p>
     * 该方法使用反射机制从当前类（继承自 BaseMapperPlus 类）的泛型参数中获取第一个泛型类型 V 的 Class 对象
     *
     * @return 当前类的泛型类型 V 的 Class 对象
     */
    default Class<V> currentDtoClass() {
        return (Class<V>) ReflectionKit.getSuperClassGenericType(this.getClass(), BaseMapperPlus.class, 1);
    }

    /**
     * 获取当前类的泛型类型 T 的 Class 对象
     * <p>
     * 该方法使用反射机制从当前类（继承自 BaseMapperPlus 类）的泛型参数中获取第一个泛型类型 T 的 Class 对象
     *
     * @return 当前类的泛型类型 T 的 Class 对象
     */
    default Class<T> currentModelClass() {
        return (Class<T>) ReflectionKit.getSuperClassGenericType(this.getClass(), BaseMapperPlus.class, 0);
    }

    /**
     * 使用默认的查询条件查询并返回结果列表
     *
     * @return 返回查询结果的列表
     */
    default List<T> selectList() {
        return this.selectList(new QueryWrapper<>());
    }

    /**
     * 批量插入实体对象集合
     *
     * @param entityList 实体对象集合
     * @return 插入操作是否成功的布尔值
     */
    default boolean batchInsert(Collection<T> entityList) {
        return Db.saveBatch(entityList);
    }

    /**
     * 批量插入实体对象集合并指定批处理大小
     *
     * @param entityList 实体对象集合
     * @param batchSize  批处理大小
     * @return 插入操作是否成功的布尔值
     */
    default boolean batchInsert(Collection<T> entityList, int batchSize) {
        return Db.saveBatch(entityList, batchSize);
    }

    /**
     * 批量根据ID更新实体对象集合
     *
     * @param entityList 实体对象集合
     * @return 更新操作是否成功的布尔值
     */
    default boolean batchUpdateById(Collection<T> entityList) {
        return Db.updateBatchById(entityList);
    }

    /**
     * 批量根据ID更新实体对象集合并指定批处理大小
     *
     * @param entityList 实体对象集合
     * @param batchSize  批处理大小
     * @return 更新操作是否成功的布尔值
     */
    default boolean batchUpdateById(Collection<T> entityList, int batchSize) {
        return Db.updateBatchById(entityList, batchSize);
    }

    /**
     * 批量插入或更新实体对象集合
     *
     * @param entityList 实体对象集合
     * @return 插入或更新操作是否成功的布尔值
     */
    default boolean insertOrUpdateBatch(Collection<T> entityList) {
        return Db.saveOrUpdateBatch(entityList);
    }

    /**
     * 批量插入或更新实体对象集合并指定批处理大小
     *
     * @param entityList 实体对象集合
     * @param batchSize  批处理大小
     * @return 插入或更新操作是否成功的布尔值
     */
    default boolean insertOrUpdateBatch(Collection<T> entityList, int batchSize) {
        return Db.saveOrUpdateBatch(entityList, batchSize);
    }

    /**
     * 根据ID查询单个DTO对象
     *
     * @param id 主键ID
     * @return 查询到的单个DTO对象
     */
    default V selectDtoById(Serializable id) {
        return selectDtoById(id, this.currentDtoClass());
    }

    /**
     * 根据ID查询单个DTO对象并将其转换为指定的DTO类
     *
     * @param id      主键ID
     * @param DTOClass 要转换的DTO类的Class对象
     * @param <C>     DTO类的类型
     * @return 查询到的单个dto对象，经过转换为指定的DTO类后返回
     */
    default <C> C selectDtoById(Serializable id, Class<C> DTOClass) {
        T obj = this.selectById(id);
        if (Objects.isNull(obj)) {
            return null;
        }
        return BeanMapper.map(obj, DTOClass);
    }

    /**
     * 根据ID集合批量查询DTO对象列表
     *
     * @param idList 主键ID集合
     * @return 查询到的DTO对象列表
     */
    default List<V> selectDtoByIds(Collection<? extends Serializable> idList) {
        return selectDtoByIds(idList, this.currentDtoClass());
    }

    /**
     * 根据ID集合批量查询实体对象列表，并将其转换为指定的DTO对象列表
     *
     * @param idList  主键ID集合
     * @param DTOClass 要转换的DTO类的Class对象
     * @param <C>     DTO类的类型
     * @return 查询到的DTO对象列表，经过转换为指定的DTO类后返回
     */
    default <C> List<C> selectDtoByIds(Collection<? extends Serializable> idList, Class<C> DTOClass) {
        List<T> list = this.selectByIds(idList);
        if (CollectionUtils.isEmpty(list)) {
            return Collections.emptyList();
        }
        return BeanMapper.mapList(list, DTOClass);
    }

    /**
     * 根据查询条件Map查询DTO对象列表
     *
     * @param map 查询条件Map
     * @return 查询到的DTO对象列表
     */
    default List<V> selectDtoByMap(Map<String, Object> map) {
        return selectDtoByMap(map, this.currentDtoClass());
    }

    /**
     * 根据查询条件Map查询实体对象列表，并将其转换为指定的DTO对象列表
     *
     * @param map     查询条件Map
     * @param DTOClass 要转换的DTO类的Class对象
     * @param <C>     DTO类的类型
     * @return 查询到的DTO对象列表，经过转换为指定的DTO类后返回
     */
    default <C> List<C> selectDtoByMap(Map<String, Object> map, Class<C> DTOClass) {
        List<T> list = this.selectByMap(map);
        if (CollectionUtils.isEmpty(list)) {
            return Collections.emptyList();
        }
        return BeanMapper.mapList(list, DTOClass);
    }

    /**
     * 根据条件查询单个DTO对象
     *
     * @param wrapper 查询条件Wrapper
     * @return 查询到的单个DTO对象
     */
    default V selectDtoOne(Wrapper<T> wrapper) {
        return selectDtoOne(wrapper, this.currentDtoClass());
    }

    /**
     * 根据条件查询单个DTO对象，并根据需要决定是否抛出异常
     *
     * @param wrapper 查询条件Wrapper
     * @param throwEx 是否抛出异常的标志
     * @return 查询到的单个DTO对象
     */
    default V selectDtoOne(Wrapper<T> wrapper, boolean throwEx) {
        return selectDtoOne(wrapper, this.currentDtoClass(), throwEx);
    }

    /**
     * 根据条件查询单个DTO对象，并指定返回的DTO对象的类型
     *
     * @param wrapper 查询条件Wrapper
     * @param DtoClass 返回的DTO对象的Class对象
     * @param <C>     返回的DTO对象的类型
     * @return 查询到的单个DTO对象，经过类型转换为指定的DTO类后返回
     */
    default <C> C selectDtoOne(Wrapper<T> wrapper, Class<C> DtoClass) {
        T obj = this.selectOne(wrapper);
        if (Objects.isNull(obj)) {
            return null;
        }
        return BeanMapper.map(obj, DtoClass);
    }

    /**
     * 根据条件查询单个实体对象，并将其转换为指定的DTO对象
     *
     * @param wrapper 查询条件Wrapper
     * @param DTOClass 要转换的DTO类的Class对象
     * @param throwEx 是否抛出异常的标志
     * @param <C>     DTO类的类型
     * @return 查询到的单个DTO对象，经过转换为指定的DTO类后返回
     */
    default <C> C selectDtoOne(Wrapper<T> wrapper, Class<C> DTOClass, boolean throwEx) {
        T obj = this.selectOne(wrapper, throwEx);
        if (Objects.isNull(obj)) {
            return null;
        }
        return BeanMapper.map(obj, DTOClass);
    }

    /**
     * 查询所有DTO对象列表
     *
     * @return 查询到的DTO对象列表
     */
    default List<V> selectDtoList() {
        return selectDtoList(new QueryWrapper<>(), this.currentDtoClass());
    }

    /**
     * 根据条件查询DTO对象列表
     *
     * @param wrapper 查询条件Wrapper
     * @return 查询到的DTO对象列表
     */
    default List<V> selectDtoList(Wrapper<T> wrapper) {
        return selectDtoList(wrapper, this.currentDtoClass());
    }

    /**
     * 根据条件查询实体对象列表，并将其转换为指定的DTO对象列表
     *
     * @param wrapper 查询条件Wrapper
     * @param DTOClass 要转换的DTO类的Class对象
     * @param <C>     DTO类的类型
     * @return 查询到的DTO对象列表，经过转换为指定的DTO类后返回
     */
    default <C> List<C> selectDtoList(Wrapper<T> wrapper, Class<C> DTOClass) {
        List<T> list = this.selectList(wrapper);
        if (CollectionUtils.isEmpty(list)) {
            return Collections.emptyList();
        }
        return BeanMapper.mapList(list, DTOClass);
    }

    /**
     * 根据条件分页查询DTO对象列表
     *
     * @param page    分页信息
     * @param wrapper 查询条件Wrapper
     * @return 查询到的DTO对象分页列表
     */
    default <P extends IPage<V>> P selectDtoPage(IPage<T> page, Wrapper<T> wrapper) {
        return selectDtoPage(page, wrapper, this.currentDtoClass());
    }

    /**
     * 根据条件分页查询实体对象列表，并将其转换为指定的DTO对象分页列表
     *
     * @param page    分页信息
     * @param wrapper 查询条件Wrapper
     * @param DTOClass 要转换的DTO类的Class对象
     * @param <C>     DTO类的类型
     * @param <P>     DTO对象分页列表的类型
     * @return 查询到的DTO对象分页列表，经过转换为指定的DTO类后返回
     */
    default <C, P extends IPage<C>> P selectDtoPage(IPage<T> page, Wrapper<T> wrapper, Class<C> DTOClass) {
        // 根据条件分页查询实体对象列表
        List<T> list = this.selectList(page, wrapper);
        // 创建一个新的DTO对象分页列表，并设置分页信息
        IPage<C> dtoPage = new Page<>(page.getCurrent(), page.getSize(), page.getTotal());
        if (CollectionUtils.isEmpty(list)) {
            return (P) dtoPage;
        }
        dtoPage.setRecords(BeanMapper.mapList(list, DTOClass));
        return (P) dtoPage;
    }

    /**
     * 根据条件查询符合条件的对象，并将其转换为指定类型的对象列表
     *
     * @param wrapper 查询条件Wrapper
     * @param mapper  转换函数，用于将查询到的对象转换为指定类型的对象
     * @param <C>     要转换的对象的类型
     * @return 查询到的符合条件的对象列表，经过转换为指定类型的对象后返回
     */
    default <C> List<C> selectObjs(Wrapper<T> wrapper, Function<? super Object, C> mapper) {
        return this.selectObjs(wrapper).stream().filter(Objects::nonNull).map(mapper).collect(Collectors.toList());
    }

}
